import torch
import h5py
f = h5py.File('out/feature/test.h5', 'r')
model = torch.jit.load('out/models/traced.pt')

x = torch.Tensor(f['x'][0]).unsqueeze(0)
x = torch.ones_like(x)
y = model(x)
print(y)
